package org.jeecg.modules.basedata.controller;

import cn.jiguang.common.utils.StringUtils;
import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.beust.jcommander.internal.Lists;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.jeecg.common.api.vo.Result;
import org.jeecg.common.aspect.annotation.AutoLog;
import org.jeecg.common.system.query.QueryGenerator;
import org.jeecg.common.system.vo.LoginUser;
import org.jeecg.common.util.SsoUtil;
import org.jeecg.common.util.oConvertUtils;
import org.jeecg.entity.basedata.BaseFeeRuleContract;
import org.jeecg.entity.basedata.RentContract;
import org.jeecg.modules.basedata.service.IBaseFeeRuleContractService;
import org.jeecg.modules.basedata.service.IRentContractService;
import org.jeecg.modules.system.service.ISysUserRoleService;
import org.jeecgframework.poi.excel.ExcelImportUtil;
import org.jeecgframework.poi.excel.def.NormalExcelConstants;
import org.jeecgframework.poi.excel.entity.ExportParams;
import org.jeecgframework.poi.excel.entity.ImportParams;
import org.jeecgframework.poi.excel.view.JeecgEntityExcelView;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.util.Arrays;
import java.util.List;
import java.util.Map;

/**
 * @Description: 收费标准合同关联表
 * @Author: liurui
 * @Date:   2021-07-12
 * @Version: V1.0
 */
@Slf4j
@Api(tags="收费标准合同关联表")
@RestController
@RequestMapping("/b/basedata/baseFeeRuleContract")
public class BaseFeeRuleContractController {
	@Autowired
	private IBaseFeeRuleContractService baseFeeRuleContractService;
	@Autowired
    private ISysUserRoleService sysUserRoleService;
	@Autowired
	private IRentContractService rentContractService;
	
	/**
	  * 分页列表查询
	 * @param baseFeeRuleContract
	 * @param pageNo
	 * @param pageSize
	 * @param req
	 * @return
	 */
	@AutoLog(value = "收费标准合同关联表-分页列表查询")
	@ApiOperation(value="收费标准合同关联表-分页列表查询", notes="收费标准合同关联表-分页列表查询")
	@GetMapping(value = "/list")
	public Result<IPage<BaseFeeRuleContract>> queryPageList(BaseFeeRuleContract baseFeeRuleContract,
									  @RequestParam(name="pageNo", defaultValue="1") Integer pageNo,
									  @RequestParam(name="pageSize", defaultValue="10") Integer pageSize,
									  HttpServletRequest req) {
		Result<IPage<BaseFeeRuleContract>> result = new Result<IPage<BaseFeeRuleContract>>();
		baseFeeRuleContract.setSqlMap(sysUserRoleService.dataScopeProjectOnlyFilter(SsoUtil.getLoginUser(),"p",baseFeeRuleContract.getMenuPath(),null,baseFeeRuleContract.getDepartIdEx()));
		Page<BaseFeeRuleContract> page = new Page<BaseFeeRuleContract>(pageNo, pageSize);
		//page.setRecords(baseFeeRuleContractService.findPage(page,baseFeeRuleContract));
		result.setSuccess(true);
		result.setResult(page);
		return result;
	}
	
	/**
	  *   添加
	 * @param baseFeeRuleContract
	 * @return
	 */
	@AutoLog(value = "收费标准合同关联表-添加")
	@ApiOperation(value="收费标准合同关联表-添加", notes="收费标准合同关联表-添加")
	@PostMapping(value = "/add")
	public Result<BaseFeeRuleContract> add(@RequestBody BaseFeeRuleContract baseFeeRuleContract) {
		Result<BaseFeeRuleContract> result = new Result<BaseFeeRuleContract>();
		try {
			baseFeeRuleContractService.save(baseFeeRuleContract);
			result.success("添加成功！");
		} catch (Exception e) {
			log.error(e.getMessage(),e);
			result.error500("操作失败");
		}
		return result;
	}

	 /**
	  * @param flag 接口标识:0：批量 1：所有
	  */
	 @RequestMapping("bindBatch")
	 public Result<?> bindBatch( String rentIds,@RequestParam("feeId") String feeId,@RequestParam("feeRuleId")String feeRuleId,
								 Integer flag,String projectId){
		 Result<Object> result = new Result<>();
		 List<RentContract> rentList = Lists.newArrayList();
		 String userId = SsoUtil.getLoginUser().getId();
		 if (flag == 1 ){
			 if (StringUtils.isEmpty(projectId)){
				 return Result.error("请选择项目");
			 }
			 //获取该项目下所有合同
			 rentList = rentContractService.getAllContractInfo(projectId);
		 }else {
			 //单个设置 或批量设置
			 List<String> rentIdList = Arrays.asList(rentIds.split(";"));
			 rentList = rentContractService.getContractInfo(rentIdList,feeId);
		 }
		 List<BaseFeeRuleContract> cfrList = Lists.newArrayList();
		 for (RentContract rentContract:rentList){
			 BaseFeeRuleContract cfr = new BaseFeeRuleContract();
			 cfr.setFloorId(rentContract.getFloorId());
			 cfr.setContractId(rentContract.getId());
			 cfr.setProjectId(rentContract.getProjectId());
			 cfr.setFeeId(feeId);
			 cfr.setRoomId(rentContract.getRoomId());
			 cfr.setFeeRuleId(feeRuleId);
			 cfr.setCreateBy(userId);
			 cfrList.add(cfr);
		 }
		 try {
			 baseFeeRuleContractService.saveBatch(cfrList);
			 result.setSuccess(true);
			 result.setMessage("收费项设置成功");
		 }catch (Exception e){
			 log.info(e.getMessage());
			 return result.error500("系统错误,请联系管理员");
		 }
		 return result;
	 }
	/**
	  *  编辑
	 * @param baseFeeRuleContract
	 * @return
	 */
	@AutoLog(value = "收费标准合同关联表-编辑")
	@ApiOperation(value="收费标准合同关联表-编辑", notes="收费标准合同关联表-编辑")
	@PutMapping(value = "/edit")
	public Result<BaseFeeRuleContract> edit(@RequestBody BaseFeeRuleContract baseFeeRuleContract) {
		Result<BaseFeeRuleContract> result = new Result<BaseFeeRuleContract>();
		BaseFeeRuleContract baseFeeRuleContractEntity = baseFeeRuleContractService.getById(baseFeeRuleContract.getId());
		if(baseFeeRuleContractEntity==null) {
			result.error500("未找到对应实体");
		}else {
			boolean ok = baseFeeRuleContractService.updateById(baseFeeRuleContract);
			//TODO 返回false说明什么？
			if(ok) {
				result.success("修改成功!");
			}
		}
		
		return result;
	}
	
	/**
	  *   通过id删除
	 * @param id
	 * @return
	 */
	@AutoLog(value = "收费标准合同关联表-通过id删除")
	@ApiOperation(value="收费标准合同关联表-通过id删除", notes="收费标准合同关联表-通过id删除")
	@DeleteMapping(value = "/delete")
	public Result<?> delete(@RequestParam(name="id",required=true) String id) {
		try {
			baseFeeRuleContractService.removeById(id);
		} catch (Exception e) {
			log.error("删除失败",e.getMessage());
			return Result.error("删除失败!");
		}
		return Result.ok("删除成功!");
	}
	
	/**
	  *  批量删除
	 * @param ids
	 * @return
	 */
	@AutoLog(value = "收费标准合同关联表-批量删除")
	@ApiOperation(value="收费标准合同关联表-批量删除", notes="收费标准合同关联表-批量删除")
	@DeleteMapping(value = "/deleteBatch")
	public Result<BaseFeeRuleContract> deleteBatch(@RequestParam(name="ids",required=true) String ids) {
		Result<BaseFeeRuleContract> result = new Result<BaseFeeRuleContract>();
		if(ids==null || "".equals(ids.trim())) {
			result.error500("参数不识别！");
		}else {
			this.baseFeeRuleContractService.removeByIds(Arrays.asList(ids.split(",")));
			result.success("删除成功!");
		}
		return result;
	}
	
	/**
	  * 通过id查询
	 * @param id
	 * @return
	 */
	@AutoLog(value = "收费标准合同关联表-通过id查询")
	@ApiOperation(value="收费标准合同关联表-通过id查询", notes="收费标准合同关联表-通过id查询")
	@GetMapping(value = "/queryById")
	public Result<BaseFeeRuleContract> queryById(@RequestParam(name="id",required=true) String id) {
		Result<BaseFeeRuleContract> result = new Result<BaseFeeRuleContract>();
		BaseFeeRuleContract baseFeeRuleContract = baseFeeRuleContractService.getById(id);
		if(baseFeeRuleContract==null) {
			result.error500("未找到对应实体");
		}else {
			result.setResult(baseFeeRuleContract);
			result.setSuccess(true);
		}
		return result;
	}

  /**
      * 导出excel
   *
   * @param request
   * @param response
   */
  @RequestMapping(value = "/exportXls")
  public ModelAndView exportXls(HttpServletRequest request, HttpServletResponse response) {
      // Step.1 组装查询条件
      QueryWrapper<BaseFeeRuleContract> queryWrapper = null;
      try {
          String paramsStr = request.getParameter("paramsStr");
          if (oConvertUtils.isNotEmpty(paramsStr)) {
              String deString = URLDecoder.decode(paramsStr, "UTF-8");
              BaseFeeRuleContract baseFeeRuleContract = JSON.parseObject(deString, BaseFeeRuleContract.class);
              queryWrapper = QueryGenerator.initQueryWrapper(baseFeeRuleContract, request.getParameterMap());
          }
      } catch (UnsupportedEncodingException e) {
          e.printStackTrace();
      }
      LoginUser loginUser =  SsoUtil.getLoginUser();
      //Step.2 AutoPoi 导出Excel
      ModelAndView mv = new ModelAndView(new JeecgEntityExcelView());
      List<BaseFeeRuleContract> pageList = baseFeeRuleContractService.list(queryWrapper);
      //导出文件名称
      mv.addObject(NormalExcelConstants.FILE_NAME, "收费标准合同关联表列表");
      mv.addObject(NormalExcelConstants.CLASS, BaseFeeRuleContract.class);
      mv.addObject(NormalExcelConstants.PARAMS, new ExportParams("收费标准合同关联表列表数据", "导出人:" + loginUser.getRealname(), "导出信息"));
      mv.addObject(NormalExcelConstants.DATA_LIST, pageList);
      return mv;
  }

  /**
      * 通过excel导入数据
   *
   * @param request
   * @param response
   * @return
   */
  @RequestMapping(value = "/importExcel", method = RequestMethod.POST)
  public Result<?> importExcel(HttpServletRequest request, HttpServletResponse response) {
      MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
      Map<String, MultipartFile> fileMap = multipartRequest.getFileMap();
      for (Map.Entry<String, MultipartFile> entity : fileMap.entrySet()) {
          MultipartFile file = entity.getValue();// 获取上传文件对象
          ImportParams params = new ImportParams();
          params.setTitleRows(2);
          params.setHeadRows(1);
          params.setNeedSave(true);
          try {
              List<BaseFeeRuleContract> listBaseFeeRuleContracts = ExcelImportUtil.importExcel(file.getInputStream(), BaseFeeRuleContract.class, params);
              baseFeeRuleContractService.saveBatch(listBaseFeeRuleContracts);
              return Result.ok("文件导入成功！数据行数:" + listBaseFeeRuleContracts.size());
          } catch (Exception e) {
              log.error(e.getMessage(),e);
              return Result.error("文件导入失败:"+e.getMessage());
          } finally {
              try {
                  file.getInputStream().close();
              } catch (IOException e) {
                  e.printStackTrace();
              }
          }
      }
      return Result.ok("文件导入失败！");
  }

}
